from ray.tune.registry import register_env

from rl_evolve.envs.uav_2d import RasterWrapper, UavEnvironment

register_env("UavEnvironment", lambda cfg: RasterWrapper(UavEnvironment(**cfg)))
